package com.myz.rabbitmq.study4;

import com.myz.rabbitmq.utils.RabbitMqUtils;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;

import java.util.UUID;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;

/*
* 练习发布确认模式
* 通过比较使用的时间，来了解哪种发布模式更好
* 1、单个确认模式 发布1000条单独确认消息,耗时:1205ms
* 2、批量确认模式 发布1000条批量确认消息,耗时:192ms
* 3、异步批量确认 发布1000条异步确认消息,耗时:73ms
*    加入错误监听的异步确认 发布1000条异步确认消息,耗时:1192ms
* */
public class ConfirmMessage {
    // 批量发消息的个数
    public static final int MESSAGE_NUM = 1000;

    public static void main(String[] args) throws Exception {
        // 单个确认
//        publicMessage();
        // 批量确认
//        publicMessages();
        // 异步确认
        publicMessagesAsync();
    }

    // 单个确认
    public static void publicMessage() throws Exception {
        // 获取信道
        Channel channel = RabbitMqUtils.getChannel();
        // 队列名称使用uuid
        String queueName = UUID.randomUUID().toString();
        // 声明队列 参数：队列名、是否持久化、是否只允许一个消费者消费、是否自动删除、其他参数
        channel.queueDeclare(queueName,true,false,false,null);
        // 开启发布确认
        channel.confirmSelect();
        // 计算时间
        long begin = System.currentTimeMillis();
        // 批量发消息
        for (int i = 0; i < MESSAGE_NUM; i++) {
            String message = i + "";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 单个消息就马上进行发布确认
            boolean flag = channel.waitForConfirms();
            if (flag) {
                System.out.println("消息发送成功:"+message);
            }
        }
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_NUM+"条单独确认消息,耗时:"+(end-begin)+"ms");
    }

    // 批量确认
    public static void publicMessages() throws Exception {
        // 获取信道
        Channel channel = RabbitMqUtils.getChannel();
        // 队列名称使用uuid
        String queueName = UUID.randomUUID().toString();
        // 声明队列 参数：队列名、是否持久化、是否只允许一个消费者消费、是否自动删除、其他参数
        channel.queueDeclare(queueName,true,false,false,null);
        // 开启发布确认
        channel.confirmSelect();
        // 计算时间
        long begin = System.currentTimeMillis();
        // 批量确认消息规模的大小
        int batchSize = 100;
        // 批量发消息
        for (int i = 1; i <= MESSAGE_NUM; i++) {
            String message = i + "";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 判断当发送消息达到100条时 批量确认一次
            if (i%batchSize == 0) {
                channel.waitForConfirms();
            }
        }
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_NUM+"条批量确认消息,耗时:"+(end-begin)+"ms");
    }

    // 异步确认
    public static void publicMessagesAsync() throws Exception {
        // 获取信道
        Channel channel = RabbitMqUtils.getChannel();
        // 队列名称使用uuid
        String queueName = UUID.randomUUID().toString();
        // 声明队列 参数：队列名、是否持久化、是否只允许一个消费者消费、是否自动删除、其他参数
        channel.queueDeclare(queueName,true,false,false,null);
        // 开启发布确认
        channel.confirmSelect();
        /*
        * 线程安全有序的一个哈希表 适用于高并发的情况
        * 1.轻松的将序号与消息进行关联
        * 2.轻松批量删除条目 只要给到序号
        * 3.支持高并发(多线程)
         */
        ConcurrentSkipListMap<Long,String> map = new ConcurrentSkipListMap<>();
        // 消息接收成功的回调函数 参数1:消息标识 参数2:是否为批量确认
        ConfirmCallback ackCallback = (deliveryTag,multiple) -> {
            // 2.删除已经确认的消息 剩下的就是未确认的消息
            if (multiple) { // 如果是批量
                ConcurrentNavigableMap<Long, String> confirmed = map.headMap(deliveryTag);
                confirmed.clear();
            } else {
                map.remove(deliveryTag);
            }
        };
        // 消息接收失败的回调函数 参数1:消息标识 参数2:是否为批量确认
        ConfirmCallback nackCallback = (deliveryTag,multiple) -> {
            // 3.打印一下未确认的消息有哪些
            String errorMessage = map.get(deliveryTag);
            System.out.println("未确认的消息:"+errorMessage+"  未确认的消息编号:"+deliveryTag);
        };
        // 准备消息的监听器 监听哪些消息成功了 哪些消息失败了
        /*1.监听哪些消息成功了 2.监听哪些消息失败了*/
        channel.addConfirmListener(ackCallback,nackCallback); // 异步通知
        // 计算时间
        long begin = System.currentTimeMillis();
        for (int i = 0; i < MESSAGE_NUM; i++) {
            String message = i + "";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 1.此处记录所有要发送的消息 通过channel.getNextPublishSeqNo()方法获取消息的编号作为key传入map
            map.put(channel.getNextPublishSeqNo(),message);
        }
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_NUM+"条异步确认消息,耗时:"+(end-begin)+"ms");
    }
}
